from llama_index.core.base.base_retriever import BaseRetriever
from llama_index.retrievers.mongodb_atlas_bm25_retriever.base import (
    MongoDBAtlasBM25Retriever,
)


def test_class():
    names_of_base_classes = [b.__name__ for b in MongoDBAtlasBM25Retriever.__mro__]
    assert BaseRetriever.__name__ in names_of_base_classes
